package com.atues.utils;


import com.aliyun.oss.ClientConfiguration;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.*;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.util.StringUtils;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLDecoder;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @Author pc
 * @create 2024/8/6 14:57
 */
@Component
public class OssUtil {
    protected static Log logger = LogFactory.getLog(OssUtil.class);

    // yourEndpoint填写Bucket所在地域对应的Endpoint。
    final static String endpoint = OssConstantPropertiesUtil.END_POIND;
    final static String accessKeyId = OssConstantPropertiesUtil.ACCESS_KEY_ID;
    final static String accessKeySecret = OssConstantPropertiesUtil.ACCESS_KEY_SECRET;
    final static String bucketName = OssConstantPropertiesUtil.BUCKET_NAME;
    final static String rootpath = "blog/";

    private static OSSClient ossclient=null;

    /**
     * 获取阿里云OSS客户端对象
     *
     * return ossClient
     */
    static{
        //设置超时机制和重试机制
        ClientConfiguration conf = new ClientConfiguration();
        conf.setConnectionTimeout(5000);
        conf.setMaxErrorRetry(3);
        ossclient = new OSSClient(endpoint, accessKeyId, accessKeySecret,conf);
    }


    /**
     * 创建存储空间
     *
     * param ossClient OSS连接
     * param bucketName 存储空间
     * return
     */
    public static String createBucketName(String bucketName) {
        // 存储空间
        final String bucketNames = bucketName;
        if (!ossclient.doesBucketExist(bucketName)) {
            // 创建存储空间
            Bucket bucket = ossclient.createBucket(bucketName);
            return bucket.getName();
        }
        return bucketNames;
    }


    //将文件上传到阿里云上
    public static String uploadFileToOss(MultipartFile multipartFile, String path, String fileName) {
        InputStream inputStream = null;
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
        fileName = path + fileName;
        try {
            //获取上传文件输入流
            inputStream = multipartFile.getInputStream();
            // 调用oss方法实现上传
            ossClient.putObject(bucketName, fileName, inputStream);
            // 关闭OSSClient。
            ossClient.shutdown();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return fileName;
    }
    /**
     * 批量删除文件
     *
     * @param paths 文件相对地址
     */
    public static void deleteFile(List<String> paths) {
        List<String> newPath = paths.stream().map(path -> "upload/" + path).collect(Collectors.toList());
        OSS ossClient = null;
        try {
            // 创建OSSClient实例。
            ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
            //删除
            ossClient.deleteObjects(new DeleteObjectsRequest(bucketName).withKeys(newPath).withEncodingType("url"));

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (ossClient != null) {
                ossClient.shutdown();
            }
        }
    }
    /**
     * 创建模拟文件夹
     *
     * param ossClient oss连接
     * param bucketName 存储空间
     * param folder 模拟文件夹名如"qj_nanjing/"
     * return 文件夹名
     */
    public static String createFolder(String bucketName, String folder) {
        // 文件夹名
        final String keySuffixWithSlash = folder;
        // 判断文件夹是否存在，不存在则创建
        if (!ossclient.doesObjectExist(bucketName, keySuffixWithSlash)) {
            // 创建文件夹
            ossclient.putObject(bucketName, keySuffixWithSlash, new ByteArrayInputStream(new byte[0]));
            logger.info("创建文件夹成功");
            // 得到文件夹名
            OSSObject object = ossclient.getObject(bucketName, keySuffixWithSlash);
            String fileDir = object.getKey();
            return fileDir;
        }
        return keySuffixWithSlash;
    }
    /**
     * 根据key删除OSS服务器上的文件
     *
     * param ossClient oss连接
     * param bucketName 存储空间
     * param folder 模拟文件夹名 如"qj_nanjing/"
     * param key Bucket下的文件的路径名+文件名 如："upload/cake.jpg"
     */
    public static void deleteFile(String bucketName, String folder, String key) {
        ossclient.deleteObject(bucketName, folder + key);
        logger.info("删除" + bucketName + "下的文件" + folder + key + "成功");
    }

    /**
     * 上传文件流至OSS
     *
     * param ossClient oss连接
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param bucketName 存储空间
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public static String uploadObjectOSS(String storePath,String fileName,InputStream is) {
        String resultStr = null;

        createFolder(bucketName, storePath);
        try {
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            // metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossclient.putObject(bucketName, storePath + fileName, is, metadata);
            // 解析结果
            resultStr = storePath + fileName;
            logger.info("putResult.getETag():"+putResult.getETag());
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }

    /**
     * 上传本地文件至OSS
     *
     * param ossClient oss连接
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param bucketName 存储空间
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public static String uploadObjectOSS(String storePath, File file,InputStream is) {
        String resultStr = null;
        createFolder(bucketName, storePath);
        try {
            // 文件名
            String fileName = file.getName();
            // 文件大小
            Long fileSize = file.length();
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossclient.putObject(bucketName, StringUtils.hasText(storePath)?storePath:rootpath + fileName, is, metadata);
            // 解析结果
            resultStr = storePath + fileName;
            logger.info("putResult.getETag():"+putResult.getETag());
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }

    /**
     * 上传指定文件名本地文件至OSS
     * @param fileName 指定文件名
     * param ossClient oss连接
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param bucketName 存储空间
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public static String uploadObjectOSS(String storePath,String fileName, File file,InputStream is) {
        String resultStr = null;
        createFolder(bucketName, storePath);
        try {
          /*  // 文件名
            String fileName = file.getName();*/
            // 文件大小
            Long fileSize = file.length();
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            // 上传文件 (上传文件流的形式)
            PutObjectResult putObjectResult = ossclient.putObject(bucketName, storePath + fileName, is, metadata);
            // 解析结果
            resultStr = storePath + fileName;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return resultStr;
    }

    /**
     * 通过文件名判断并获取OSS服务文件上传时文件的contentType
     *
     * param fileName 文件名
     * return 文件的contentType
     */
    public static String getContentType(String fileName) {
        // 文件的后缀名
        String fileExtension = fileName.substring(fileName.lastIndexOf("."));
        if (".bmp".equalsIgnoreCase(fileExtension)) {
            return "image/bmp";
        }
        if (".gif".equalsIgnoreCase(fileExtension)) {
            return "image/gif";
        }
        if (".jpeg".equalsIgnoreCase(fileExtension) || ".jpg".equalsIgnoreCase(fileExtension)
                || ".png".equalsIgnoreCase(fileExtension)) {
            return "image/jpeg";
        }
        if (".png".equalsIgnoreCase(fileExtension)) {
            return "image/png";
        }
        if (".html".equalsIgnoreCase(fileExtension)) {
            return "text/html";
        }
        if (".txt".equalsIgnoreCase(fileExtension)) {
            return "text/plain";
        }
        if (".vsd".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.visio";
        }
        if (".ppt".equalsIgnoreCase(fileExtension) || "pptx".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.ms-powerpoint";
        }
        if (".doc".equalsIgnoreCase(fileExtension) || "docx".equalsIgnoreCase(fileExtension)) {
            return "application/msword";
        }
        if (".xml".equalsIgnoreCase(fileExtension)) {
            return "text/xml";
        }
        // 默认返回类型
        return "";
    }

    /**
     * 下载文件到本地
     * @param ossFilePath  上传到OSS起的全路径名
     * @param filename     文件下载到本地保存的路径
     */
    public static void downloadFile(String ossFilePath, String filename) {
        ossclient.getObject(new GetObjectRequest(bucketName, ossFilePath), new File(filename));
    }

    /**
     * 流式下载文件
     * @param ossFilePath
     * @param request
     * @param response
     * @throws IOException
     */
//    public static void downloadFile(String ossFilePath, HttpServletRequest request, HttpServletResponse response) throws IOException {
//        // ossObject包含文件所在的存储空间名称、文件名称、文件元信息以及一个输入流。
//        OSSObject ossObject = ossclient.getObject(bucketName, ossFilePath);
//        String fileName = ossFilePath.substring(ossFilePath.lastIndexOf("/")+1);
//        response.setContentType("application/octet-stream; charset=utf-8");
//        String agent = request.getHeader("USER-AGENT").toLowerCase();
//        if (agent.contains("firefox")) {
//            response.setCharacterEncoding("utf-8");
//            response.setHeader("content-disposition", "attachment;filename=" + new String(fileName.getBytes(), "ISO8859-1"));
//        }else{
//            response.setHeader("Content-Disposition", "attachment; filename="+ EncodeUtils.encodeUrl(fileName));
//        }
//        //设置HTTP响应头
//        response.reset();//重置 响应头
//        response.setContentType("application/x-download");//告知浏览器下载文件，而不是直接打开，浏览器默认为打开
//        InputStream inStream = ossObject.getObjectContent();
//        byte[] b = new byte[1024];
//        int len;
//        while ((len = inStream.read(b)) > 0){
//            response.getOutputStream().write(b, 0, len);
//        }
//        inStream.close();
//        response.getOutputStream().close();
//    }

    /**
     * 获得url链接
     * @param path 文件路径
     * @return
     */
    public static String getUrl(String path, String activeTime) {
        // 设置URL过期时间为1年  3600* 1000*24*365
        long time = 3600* 1000*24*365;
        if (!StringUtils.hasText(activeTime)) {
            time = Long.parseLong(activeTime) * 60000;
        }
        Date expiration = new Date(System.currentTimeMillis() + time);
        // 生成URL
        URL url = ossclient.generatePresignedUrl(bucketName, path, expiration);
        if (url != null) {
            return url.toString();
        }
        return null;
    }

}
